python differential equation